CUDA_VISIBLE_DEVICES=2,3 python -m torch.distributed.launch --use_env --nproc_per_node=2 main.py \
--model_name "MIMO-UNet-SNN-FIRE-SCNN" \
--mode "train" \
--batch_size 1 \
--num_epoch 500 \
--save_freq 20 \
--valid_freq 50 \
--print_freq 10 \
--exp_name "fire_scnn_ts_5" \
--dist-url tcp://localhost:12345 \
--data_dir "./CamVid"